import java.util.*;

public class LevelOrder_2 {
    /**
     * 请实现一个函数按照之字形顺序打印二叉树，即第一行按照从左到右的顺序打印，
     * 第二层按照从右到左的顺序打印，第三行再按照从左到右的顺序打印，其他行以此类推。
     * @param root
     * @return
     */
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> ans = new ArrayList<List<Integer>>();
        if (root==null)
            return ans;
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.offer(root);
        boolean flag =true;
        while (!queue.isEmpty()){
            int size = queue.size();
            flag =!flag;
            List<Integer> tmp_list = new ArrayList<Integer>();
            for (int i=0;i<size;i++){
                TreeNode treeNode = queue.poll();
                tmp_list.add(treeNode.val);
                if (treeNode.left!=null){
                    queue.offer(treeNode.left);
                }
                if (treeNode.right!=null){
                    queue.offer(treeNode.right);
                }
            }
            if (flag){
                Collections.reverse(tmp_list);
            }
            ans.add(tmp_list);
        }
        return ans;
    }
}
